package cn.qingsec.satoken;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.dao.SaTokenDao;
import cn.dev33.satoken.filter.SaServletFilter;
import cn.dev33.satoken.interceptor.SaInterceptor;
import cn.dev33.satoken.jwt.StpLogicJwtForSimple;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpLogic;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult;
import org.redisson.Redisson;
import org.redisson.api.RedissonClient;
import org.redisson.client.codec.Codec;
import org.redisson.codec.JsonJacksonCodec;
import org.redisson.config.Config;
import org.redisson.config.SingleServerConfig;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;


/**
 * [Sa-Token 权限认证] 配置类
 * @author kong
 *
 */
@Configuration
public class SaTokenConfig implements WebMvcConfigurer {
	@Value("${spring.redis.host}")
	private String redisHost;

	@Value("${spring.redis.port}")
	private String redisPort;
	@ConditionalOnMissingBean
	@Primary
	@Bean
	public RedissonClient redissonClient() {
		Config config = new Config();
		// 单节点模式
		SingleServerConfig singleServerConfig = config.useSingleServer();
		singleServerConfig.setAddress("redis://"+redisHost+":"+redisPort);
		// 使用json序列化方式
		Codec codec = new JsonJacksonCodec();
		config.setCodec(codec);
		RedissonClient redissonClient = Redisson.create(config);
		return redissonClient;
	}
	@Bean
	@Primary
	public SaTokenDao saTokenDao(){
		return  new PlusSaTokenDao();
	}
	/**
	 * 注册 Sa-Token 拦截器打开注解鉴权功能
	 */
	@Override
	public void addInterceptors(InterceptorRegistry registry) {
		// 注册 Sa-Token 拦截器，打开注解鉴权功能,并校验登录状态。
		registry.addInterceptor(new SaInterceptor(handle -> StpUtil.checkLogin()))
				.addPathPatterns("/**")
				.excludePathPatterns("/login");// 排除路由
	}

    /**
     * Sa-Token 整合 jwt
     */
	@Bean
    public StpLogic getStpLogicJwt() {
    	return new StpLogicJwtForSimple();
    }
	/**
	 * Sa-Token 全局过滤器
	 */
	@Bean
	public SaServletFilter getSaServletFilter() {
		return new SaServletFilter()

				// 指定 拦截路由 与 放行路由
				.addInclude("/**").addExclude("/favicon.ico")    /* 排除掉 /favicon.ico */

				// 认证函数: 每次请求执行
				.setAuth(obj -> {
					System.out.println("[setAuth]---------- 进入Sa-Token全局认证 -----------");

					// 登录认证 -- 拦截所有路由，并排除/user/doLogin 用于开放登录
//					SaRouter.match("/**", "/user/doLogin", () -> StpUtil.checkLogin());

					// 更多拦截处理方式，请参考“路由拦截式鉴权”章节 */
				})

				// 异常处理函数：每次认证函数发生异常时执行此函数
				.setError(e -> {
					System.out.println("[setError]---------- 进入Sa-Token异常处理 -----------");
					return SaResult.error(e.getMessage());
				})

				// 前置函数：在每次认证函数之前执行
				.setBeforeAuth(r -> {
					System.out.println("[setBeforeAuth]---------- 设置一些安全响应头 ----------");
					// ---------- 设置一些安全响应头 ----------
//					SaHolder.getResponse()
//							// 服务器名称
//							.setServer("sa-server")
//							// 是否可以在iframe显示视图： DENY=不可以 | SAMEORIGIN=同域下可以 | ALLOW-FROM uri=指定域名下可以
//							.setHeader("X-Frame-Options", "SAMEORIGIN")
//							// 是否启用浏览器默认XSS防护： 0=禁用 | 1=启用 | 1; mode=block 启用, 并在检查到XSS攻击时，停止渲染页面
//							.setHeader("X-XSS-Protection", "1; mode=block")
//							// 禁用浏览器内容嗅探
//							.setHeader("X-Content-Type-Options", "nosniff")
//					;
				})
				;
	}

}
